Lenet

class Lenet : ONNXModels.CV

This model is a neural network for digit classification that take grey-scale images of digits as input and classify the major object in the image into a set of 10 different classes.

This model is just an implementation of the famous LeNet-5 model.

The model have

an input with the shape (1x1x28x28)
an output with the shape (1x10)

Constructors

Lenet

jvm

fun Lenet()

Functions

model

jvm

open fun model(modelHub: ModelHub): OnnxInferenceModel

pretrainedModel

jvm

open override fun pretrainedModel(modelHub: ModelHub): ImageRecognitionModel

Properties

inputShape

jvm

open val inputShape: LongArray?

Shape of the input accepted by this model, without batch size.

modelRelativePath

jvm

open override val modelRelativePath: String

preprocessor

jvm

open val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>